package com.huacai.collect.service.impl;

import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import com.huacai.common.utils.DateUtils;
import com.huacai.common.utils.uuid.IdUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.huacai.collect.mapper.CollectMapper;
import com.huacai.collect.domain.Collect;
import com.huacai.collect.service.ICollectService;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.util.CollectionUtils;

import static com.huacai.common.utils.SecurityUtils.getUserId;

/**
 *  博客收藏 Service业务层处理
 *
 * @author zou
 * @date 2025-05-25
 */
@Service
public class CollectServiceImpl implements ICollectService
{
    @Autowired
    private CollectMapper collectMapper;

    @Autowired
    private SqlSessionFactory sqlSessionFactory;

    /**
     * 查询 博客收藏 
     *
     * @param collectId  博客收藏 主键
     * @return  博客收藏 
     */
    @Override
    public Collect selectCollectByCollectId(String collectId)
    {
        return collectMapper.selectCollectByCollectId(collectId);
    }

    /**
     * 查询 博客收藏 列表
     *
     * @param collect  博客收藏 
     * @return  博客收藏 
     */
    @Override
    public List<Collect> selectCollectList(Collect collect)
    {
        return collectMapper.selectCollectList(collect);
    }

    /**
     * 新增 博客收藏 
     *
     * @param collect  博客收藏 
     * @return 结果
     */
    @Override
    public int insertCollect(Collect collect)
    {
        collect.setCreateTime(DateUtils.getNowDate());
        collect.setCollectId(IdUtils.randomUUID());
        collect.setUserId(getUserId());
        return collectMapper.insertCollect(collect);
    }

    /**
     * 批量新增 博客收藏 
     *
     * @param collects  博客收藏 List
     * @return 结果
     */
    @Override
    public int batchInsertCollect(List<Collect> collects)
    {
        SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false);
        int count = 0;
        if (!CollectionUtils.isEmpty(collects)) {
            try {
                for (int i = 0; i < collects.size(); i++) {
                    int row = collectMapper.insertCollect(collects.get(i));
                    // 防止内存溢出，每100次提交一次,并清除缓存
                    boolean bool = (i >0 && i%100 == 0) || i == collects.size() - 1;
                    if (bool){
                        sqlSession.commit();
                        sqlSession.clearCache();
                    }
                    count = i + 1;
                }
            }catch (Exception e){
                e.printStackTrace();
                // 没有提交的数据可以回滚
                sqlSession.rollback();
            }finally {
                sqlSession.close();
                return count;
            }
        }
        return count;
    }

    /**
     * 修改 博客收藏 
     *
     * @param collect  博客收藏 
     * @return 结果
     */
    @Override
    public int updateCollect(Collect collect)
    {
        return collectMapper.updateCollect(collect);
    }

    /**
     * 批量删除 博客收藏 
     *
     * @param collectIds 需要删除的 博客收藏 主键
     * @return 结果
     */
    @Override
    public int deleteCollectByCollectIds(String[] collectIds)
    {
        return collectMapper.deleteCollectByCollectIds(collectIds);
    }

    /**
     * 删除 博客收藏 信息
     *
     * @param collectId  博客收藏 主键
     * @return 结果
     */
    @Override
    public int deleteCollectByCollectId(String collectId)
    {
        return collectMapper.deleteCollectByCollectId(collectId);
    }

    /**
     * 根据博客ID查询某个博客被收藏的次数
     *
     * @param blogId  博客收藏 主键
     * @return 结果
     */
    @Override
    public int selectCollectCount(String blogId) {
        return collectMapper.selectCollectCount(blogId);
    }

    /**
     * 批量查询收藏数
     *
     */
    @Override
    public Map<String, Integer> batchSelectCollectCount(List<String> blogIds) {
        if (blogIds == null || blogIds.isEmpty()) {
            return Collections.emptyMap();
        }
        // 分段处理500个ID
        int batchSize = 500;
        Map<String, Integer> finalResult = new HashMap<>();
        for (int i = 0; i < blogIds.size(); i += batchSize) {
            int end = Math.min(i + batchSize, blogIds.size());
            List<String> batchIds = blogIds.subList(i, end);
            Map<String, Map<String, Object>> resultMap = collectMapper.batchSelectCollectCount(batchIds);
            resultMap.forEach((blogId, data) ->
                    finalResult.put(blogId, ((Number) data.get("collectCount")).intValue())
            );
        }
        return finalResult;

//        // 调用批量查询
//        Map<String, Map<String, Object>> resultMap = collectMapper.batchSelectCollectCount(blogIds);
//
//        // 转换成Map<blogId, collectCount>格式
//        return resultMap.entrySet().stream()
//                .collect(Collectors.toMap(
//                        Map.Entry::getKey,
//                        entry -> ((Number) entry.getValue().get("collectCount")).intValue()
//                ));
    }
}
